import mongoose, { Model } from 'mongoose';
import { ITaskSession, IMessage, ITaskMetadata, IAttachment, IUserAction } from './types';
import { TaskSessionSchema, MessageSchema, TaskMetadataSchema, AttachmentSchema, UserActionSchema } from './schemas';

// 统一导出所有Model
export const TaskSession: Model<ITaskSession> = mongoose.model<ITaskSession>('TaskSession', TaskSessionSchema);
export const Message: Model<IMessage> = mongoose.model<IMessage>('Message', MessageSchema);
export const TaskMetadata: Model<ITaskMetadata> = mongoose.model<ITaskMetadata>('TaskMetadata', TaskMetadataSchema);
export const Attachment: Model<IAttachment> = mongoose.model<IAttachment>('Attachment', AttachmentSchema);
export const UserAction: Model<IUserAction> = mongoose.model<IUserAction>('UserAction', UserActionSchema);
